Pattern with Numbers in Python